tg-me.com/ds_interview_lib/124
Last Update:
👩💻 Решение задачи на Python про сумму диагоналей матрицы
Задача: Дана матрица mat. Напишите функцию, которая возвращает сумму элементов диагоналей матрицы.
Первым делом следует уточнить у интервьюера, какие есть ограничения. Например, он сказал, что на вход мы получаем только квадратную матрицу, элементы на пересечении диагоналей считаются только один раз, элементы могут принимать значения от 1 до 100.
Тогда решение может выглядеть так:
🧩 Мы будем проходиться в цикле и сразу прибавлять к результату значения матрицы по индексам i (mat[i][i]). Дальше будем смотреть: если i не равно n-i-1, где n — размер матрицы, значит мы прибавляем к результату mat[i][n-i-1]. Эта проверка позволит нам избежать повторного прибавления к сумме элементов, которые стоят на пересечении диагоналей.
Это будет решение за O(n), так как мы один раз проходимся в цикле по элементам матрицы.
#junior
#middle
BY Библиотека собеса по Data Science | вопросы с собеседований
Share with your friend now:
tg-me.com/ds_interview_lib/124